package com.gao.file;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.function.BiConsumer;

/**
 * @Author: ShanYuqing
 * @Version: v1.0
 */
public class FileTest4 {
    /*
            键盘录入一个文件夹路径，计算每种文件的个数
     */
    public static void main(String[] args) {
         File file = FileTest.getFile();
         getOne(file);
        map.forEach((s, integer) -> System.out.println(s+":"+integer+"个"));
    }

    static Map<String, Integer> map = new HashMap<>();

    public static void getOne(File dir) {
        File[] files = dir.listFiles();
        for (File file : files) {
            if (file.isFile()) {
                String name = file.getName();
                if (name.contains(".")) {
                    String[] split = name.split("\\.");
                    if (!map.containsKey(split[split.length - 1])) {
                        map.put(split[split.length - 1], 1);
                    } else {
                        map.put(split[split.length - 1], map.get(split[split.length - 1]) + 1);
                    }
                } else {
                    if (file.listFiles() != null) {
                        getOne(file);
                    }
                }
            }
        }
    }
}
